/*
 * Copyright 2022 The OpenYurt Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package types

import (
	"net"
)

type Route struct {
	Dst     *net.IPNet
	Gateway net.IP
	MTU     int
}

func EnsureRoutes(endpointMap map[string]*Endpoint, gw net.IP) map[string]Route {
	routes := make(map[string]Route)
	for _, ep := range endpointMap {
		for _, subnet := range ep.Subnets {
			_, cidr, _ := net.ParseCIDR(subnet)
			routes[subnet] = Route{
				Dst:     cidr,
				Gateway: gw,
			}
		}
	}
	return routes
}
